package com.xxxx.server.controller;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.xxxx.server.pojo.*;
import com.xxxx.server.service.IMenuRoleService;
import com.xxxx.server.service.IMenuService;
import com.xxxx.server.service.IPositionService;
import com.xxxx.server.service.IRoleService;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDateTime;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 权限组
 * 角色的管理
 */
@RestController
@RequestMapping("/system/basic/permiss")
public class PermissController {
    @Autowired
    private IRoleService roleService;
    @Autowired
    private IMenuService menuService;
    @Autowired
    private IMenuRoleService menuRoleService;

    @ApiOperation("获取所有角色信息")
    @GetMapping("/")
    public List<Role> getAllRoles(){
        return roleService.list();
    }

    @ApiOperation("添加角色信息")
    @PostMapping("/")
    public RespBean addRole(@RequestBody Role role){
        if(!role.getName().startsWith("ROLE_")){
            role.setName("ROLE_"+role.getName());
        }
        if(roleService.save(role)){
            return RespBean.success("添加成功");
        }
        return RespBean.error("添加失败");
    }

    @ApiOperation("删除职位信息")
    @DeleteMapping("/role/{rid}")
    public RespBean delRole(@PathVariable Integer rid){
        if(roleService.removeById(rid)){
            return RespBean.success("删除成功");
        }
        return RespBean.error("删除失败");
    }

    @ApiOperation("权限组：根据角色获取三级菜单")
    @GetMapping("/menus")
    public List<Menu> getAllMenus(){
        return menuService.getAllMenus();
    }

    @ApiOperation("根据角色id（rid）查询菜单id")
    @GetMapping("/mid/{rid}")
    // 因为我们吧所有的菜单信息都查出来了，所以到时候直接获取id就可以获取信息，权限组中的勾选功能
    public List<Integer> getMidByRid(@PathVariable Integer rid){
        // 首先list中要有条件，选择list(QueryWrapper)，条件就是rid，根据角色id（rid）查询菜单id，本质上就是查menu_role这张表。
        // list(QueryWrapper)返回的是一个对象List<MenuRole>，我们需要的是一个id，使用stream.map获取mid，再用collection转为List<Integer>
        return menuRoleService
                .list(new QueryWrapper<MenuRole>().eq("rid",rid))
                .stream()
                .map(MenuRole::getMid)
                .collect(Collectors.toList());
    }

    @ApiOperation("更新角色菜单")
    @PutMapping("/")
    // 两个参数：1、角色 rid  2、菜单 mids 数组，因为一个角色对应多个菜单
    public RespBean updateRoleMenu(Integer rid,Integer[] mids ){
        return menuRoleService.updateRoleMenu(rid,mids);
    }

}
